package cn.edu.xjtu.carlWay.dpRepractice.wordBreak;

import java.util.ArrayList;
import java.util.List;

/**
 * @author Hydrion-QLz
 * @date 2022-05-01 10:09
 * @description https://leetcode-cn.com/problems/word-break/
 * <p>
 * 139. 单词拆分
 * 给你一个字符串 s 和一个字符串列表 wordDict 作为字典。请你判断是否可以利用字典中出现的单词拼接出 s 。
 * <p>
 * 注意：不要求字典中出现的单词全部都使用，并且字典中的单词可以重复使用。
 */
public class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        List<String> arr = new ArrayList<>();
        arr.add("apple");
        arr.add("pen");
        System.out.println(solution.wordBreak("applepenapple", arr));
    }

    public boolean wordBreak(String s, List<String> wordDict) {
        int n = s.length();
        int[] dp = new int[n + 1];
        dp[0] = 1;
        for (int i = 0; i < dp.length; i++) {
            for (String str : wordDict) {
                if (i - str.length() >= 0 && dp[i - str.length()] == 1) {
                    if (s.startsWith(str, i - str.length())) {
                        dp[i] = 1;
                    }
                }
            }
        }
        return dp[n] == 1;
    }
}
